package arithmetic.LeetCode;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.Map;
import java.util.TreeMap;

/**
 * 以数组 intervals 表示若干个区间的集合，其中单个区间为 intervals[i] = [starti, endi] 。
 * 请你合并所有重叠的区间，并返回 一个不重叠的区间数组，该数组需恰好覆盖输入中的所有区间。
 *
 * 示例 1：
 * 输入：intervals = [[1,3],[2,6],[8,10],[15,18]]
 * 输出：[[1,6],[8,10],[15,18]]
 * 解释：区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].
 *
 * 示例2：
 * 输入：intervals = [[1,4],[4,5]]
 * 输出：[[1,5]]
 * 解释：区间 [1,4] 和 [4,5] 可被视为重叠区间。
 *
 * 链接：https://leetcode.cn/problems/merge-intervals
 * @author jiangfeng on 2023/4/21
 */
public class MergeIntervals {
    public static void main(String[] args) {
        System.out.println(Arrays.deepToString(
                new Solution().merge(new int[][] {{2, 3}, {2, 2}, {3, 3}, {1, 3}, {5, 7}, {2, 2}, {4, 6}})));
    }

    static class Solution {

        public int[][] merge(int[][] intervals) {
            Arrays.sort(intervals, Comparator.comparingInt(a -> a[0]));
            int lastIndex=0;
            int[][] res = new int[intervals.length][2];
            res[0] = new int[]{intervals[0][0],intervals[0][1]};
            for(int i=1;i < intervals.length; i++){
                if(res[lastIndex][1]>=intervals[i][0]){
                    res[lastIndex][1] = Math.max(intervals[i][1],res[lastIndex][1]);
                }else {
                    res[++lastIndex]= new int[]{intervals[i][0],intervals[i][1]};
                }
            }
            return Arrays.copyOf(res,lastIndex+1);
        }
        public int[][] merge2(int[][] intervals) {
            //最好使用线段树,
            TreeMap<Integer,Integer> map = new TreeMap();
            for(int[] a:intervals){
                Integer i = map.get(a[0]);
                if(i!=null){
                    map.put(a[0],Math.max(a[1],i));
                }else{
                    map.put(a[0],a[1]);
                }
            }
            ArrayList<int[]> result = new ArrayList(intervals.length);

            result.add(new int[]{map.firstKey(),map.get(map.firstKey())});
            for(Map.Entry<Integer,Integer> e:map.entrySet()){
                int k = e.getKey();
                int v = e.getValue();
                int[] last = result.get(result.size() - 1);
                //需要合并
                if(last[1]>=k){
                    last[1]=Math.max(last[1],v);
                }else{
                    // 直接加
                    result.add(new int[]{k,v});
                }

            }
            int[][] r = new int[result.size()][2];
            int i=0;
            for(int[] a:result){
                r[i++]=a;
            }
            return r;

        }
    }
}
